Added return function