first commit of problemset 2 code