Fix code comments