Updated README and main cpp