Small comment imporvements