Fixing up setup.py