modify to the new version of dipy.viz.fos