Add ZSH completion functions

Add completion functions for ZSH.