Added a few comments in the program