Solution to problem 5 in C