Change function to return a string