added test class for JUnit