DOC: Improve comments