Enable to save an image in plot_world

[kaizu]

Allow something like the following:

w = load_world('test.h5')
viz.plot_world(w, filename='test.png')  # or output='test.png'?