Add precmd function for z