Better vi mode indicator for prompt