Improve setup.py slightly