HW4: Fixed the beginning and follow the comments

Hi Mike,

I just fixed the beginning and try to follow all your comments exactly. Can your check it again? Thank you so much.

-Mengyi