Initial code for homework2