updated some of the code comments