Added class back in

remove class option needed to be added back in when returning to original menu.