adding pset 2 and related code