Initial code

Initial code from pset7.zip