fixing tests and updating docs