TODO: Add before/after comparison.