Updated codegen.ml

Unified the output format of functions, make it more clear.