MAINT: add untested functions