Add the if statement.