add "if" statement