Rewrite setup.py