Fixed part 1 of the lab