Simplify setup.py