Add curly braces as requested in review.