Add the code for function return type check.