Added the add class button