Add support for polymorphic associations and STI