Correctly write if/else

This was really not working....