Got the plugin working