Adding JUnit TestCase