Update plugin for more arguments