I preferred the old VimScript version of this function. 

`vim.command('return')` is a bad idea.