Finish hw6
Fix hw6