Cleanup for pypi release