Instead of modifying import_module, why not just call it twice in plot.py?