Add test code for the plugin