updating from last week's in class code