Adding brackets

I wasn't sure what the coding style was. Sorry about that