Latest work on lab 2 part 3