Update write_plugin.py