Add some comments

Added some comments in the code.