Update GAN code