Update generator script