Worked on part 4