Update problem5.c

Added comments