Updated imports since of the use of setup.py