Small fixes for pypi