move test model classes to separate file