Update Generator.cpp