Proper syntax highlighting for if statements