Added some brackets around if/else statements

Hey this is great, but I thought it might be nice to have some brackets around things.