add solution to part B