Add support for plugins

Right, good catch, updated.