Working on generator script