Cleaning up for pypi