Lab 2 solution part 2