Put back code from last week's lab