New class for doing inference

I have introduced a class for doing the classification and making plots which makes the inference tidier and easier to use (imo). This is reflected in the new testing_example.ipynb notebook.