Fixed unit test
Updated documentation