Improved setup.py.