Tweak to setup.py for PyPi submission