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

little modification