Cleaning up setup.py