Added more comments, descriptions and confirmation of class
I also added a line at the end to print which class was selected.