Adding JUnit test cases