add return function