updated

done
Merge pull request #2 from jonathan-jing/testCode

updated