Updated import.py