Adding if statement