Improve setup.py for future releases