Add polimorphic models support