additionall coding for new button and finding a cleaner way to add / remove classes.