Initial code for assignment 1