Return the array directly instead of a function