Adding if statements