modify to the new version of dipy/viz/of