Adding missing curly braces