add class button
Merge remote-tracking branch 'origin/Sprint-4' into Unit-Tests