JUnit homework updated