Initial code for assignment 2