fixing tests, more docs