Add correct solution for C