Update main.cpp

syntax changes + uses enumerated values