Adding support of polimorphic models