Modify main.cpp and Add Screenshot