Lab 3 part 2, my solution