Added code for using the discriminator modules.