update docs for PyPI

It's a shame we had to fork it but that's ok. This updates the docs a bit.