added hw1 code