Adding if conditionals