add vi mode in prompt