Added missin g return function