Make sure installation works with setuptools.
Clean up setup.py imports.