Update setup for pypi submission