Added comment for some parts of program