Added brackets to if statement

Cleaner code.