updated the code comments