Add polimorphic model support