Initial code for Assignment 3