Enable saving of png figures in models.plot_slice.
Improve neighbourhood filtering.