Add generator stuff