Move relations to schema, instead of entity