Finish the first problem of hw2
Finish the third problem of hw2