Add more keybindings for vi mode