That would be great, can I add a new module in dipy.viz, say imreg_viz.py (Elef, do you agree)?