Update Generator class