Update import_modules.py