Tidied "if" statement
Merge pull request #4 from LukePurdy/master

Pulling tidying of if statement.