Changed the UI for adding classes