Added TestModel class