adding if statement