Improved setup.py for PyPi