DOC: Improving documentation