Adding if clause