Added test class to test some of the model objects