Fixed tests
Update docs