add if statements

add if statements