Add more functionality to the plugin