Fixed code and braces alignment