Adding some comments in the code