update plugin to add restraint support