Added missing curly braces