Updating code through pset5