add the return element to collect the answer to the function