added a button to add classes, needs work