worked on if statement