adding classes and test