Fix part 2 of the lab.