pset5 finally done!