Add generator code