IF statement exercise extended