Working on class selection and menu