Copy over code from assn1 to edit