Worked on Part 4, not working properly