Update functionality of generator.