Add partial completion highlighting to ZSH