main.cpp

Added some stuff in main.cpp