added unit tests to project