Cleaning up docs and pypi submission