Add tests for plugin