Updated the code comments