Clean up setup.py