finish pset2, need to add comments and check style