if statements fixed