Allow something like the following: ```python w = load_world('test.h5') viz.plot_world(w, filename='test.png') # or output='test.png'? ```