Add solution for assignment c.