Junit homework