DOC Improve code documentation