Update Part 'c' solution