Add Oh My ZSH vi-mode support