add tests and new classes to the project