changed test to use JUnit