Added pset 4, part II