homework 1 code added