Integrate upgrade into CLI