Added very small things to the plugin