Made code a bit neater

removed brackets from one line if statements