Add and fix vi mode