forgot to commit the tests and a small fix to the docs