Finish the 3th problem of hw1
Finish the front part of 4th problem of hw1