Added code generation for function calls and return statements.