Update generate script