add a simpler visualize function
revamp the unrolling steps and the latch output handling