Function --> Alias.... I was pretty stupid :D

Could just use an alias and have completion work out of the box....
I even tried to write completion, but then realised an alias was way easier and
more reliable.