Tidied "if" statement