Updated a few missed comments