Added a test for the model classes (details on facebook)