Added a test

* It just makes sure that the model is usable. Quick and dirty