Clean up vi mode indicator in prompt