Update Generator code