[Wen Hao] - task 4, add more assert.