Add tests for the plugin