working on the button, added set and get class names