Add additional lines to get autocompletion to work in zsh

@aharris88 could you take a look at this and confirm these are correct?  I'm not a zsh user.

Thanks!