Clean-up setup.py