Add `gen upgrade` task