Rework setup.py