Add an if statement