Update main.cpp

Added the Code on