add discriminator column functionality