Update replace_import.py