Upgrade edit widget method