Added a possible new version of the completion file. I don't do a massive amount of zsh scripting so I'm not sure if this is the optimal or preferred way of doing this but it works for my test cases.