fix tests
polish docs