Minor: updated some comments in the code