Minor Edit: Code finally builds and runs

Edited so that the code is able to build and compiled. Able to run.