Completed problem 3 in C