Minor updates
Updated documentation and test cases