refactored tests and added tests for working with classes
added addClass() and getClasses() functionality