Maybe add a TODO here so that in the future the sample record generated can be of a specific size?