Fixed input .cpp and .h for future modification
Merge branch 'master' of https://github.com/dadleyy/Project-NN