Add newly created classes and unit tests