Added some code for pset8