ENH: improve style