Updated unit test
Added docs