ENH: visualization improvements