Add main.cpp to test