Added some junit test cases