Tests

Add tests to the plugin