Update main.cpp

KGW Modifcation