Better setup.py script