Update main.cpp

!! SRH !!