Modified main.cpp and README.md