Add support for lexing function return types