updated some comments in the code