Added model classes and proper test class.
Added model classes and proper test class.