Updated comments
Small comment fix