ENH: Add plot_trans function

I need some help here. It doesn't seem to be working if I add `_find_trans` to `__init__.py` in forward. This way also it doesn't seem to be working.