add correct if statement