My pset 6 code