Made some changes to code comments