Started adding the plugin code but need to build out the tests