Made some updates to lab4 part 4