Code polishing based on review.