homework 1
added some comments