Add comment and if statement curly braces