DOC Better documentation of code