Converted to JUnit test case