Add else statement to class selection