Made the if a one liner