Change button into class