add return element to get the answer to the function