Implemented JUnit tests for project.