Updated setup.py import