Finished the homework of leture 5

Signed-off-by: mrahmiao <mrahmiao@gmail.com>