Change the return of function to be by reference.