Move model classes to test folder