Add vi mode the root shell also