Added some test cases and updated documentation