[MOD] Change models to be more flexible. Created relation