ENH: Minor changes