FIX/ENH: fix + improve fwd example

... fix viz bug + explain better