[Wen Hao] - Added parameter to builder.