Small comment fixes