Fixed some comments in the code