add if statement