work on if statements