Add the plugin code