Add plugin code