modify return of function