Fixing the syntax for the if/else