Add final version of lab 4 part 4