Minor modifications to submit to pypi