Small Changes to cover up with the changed made how the plugin works