JUnit homework completed