Added an elementary test to test the model