Support function param and returning types.
Decorate AST with type info - first pass.