first problem of pset1 complete.