Lab 4 solution – part 2