Fix setup and pypi installation