Homework

Fix some issue and add the test code of wenting and lihao