MAINT: Improve documentation and coverage