Addition of return to function