Improve shortcuts in insert mode