a little modification
Merge pull request #248 from COMP231-S63G1/littlemodify

a little modification