fixing the unit test
updating docs