fixed up part 2 - added more to part 4