add visualization to test_stack_and_visualize