add if statements