Added some test cases..