improved API for ds.plot