Update problem 2.5.c

Comment fix, clarity in code.