Comment to clarify what the program does

Hi, I saw your code and it was not directly clear to me what the program did before running it, so I added a comment.