Fixed tests
Added docs