Lab 3 solution part 2