DOC: Improved class-level documentation