add relationship into models. fixed yml for tests