added tests for function return.