Fixing up docs for pypi