Update problem4.c

Added comments