finished with JUnit Test Class, but has to be revised