Better if-statement