Pset 5 for stats221