MAINT: add more info to the output