Minor update and comment change