Added some code to Xtrong's code
removed unnecessary comments.
Merge pull request #1 from TonnyXu/master

Added some code to Xtrong's code