Improve setup.py.