Update solution for problem C2