Added braces for if