Created working version of plugin for zsh