Small changes, plugin is done.