Added some functionality to your plugin