MAINT: Remove and update print statements