Added yes or no to if statement