Finish the assaignment

Add some comments