Cleanup on setup.py