ENH: improved the style