Add main.cpp
Add main function