add the return() function