Can you add braces here? Same for lines 228-229.