added rest of code, successfully built on my pc and ran training on mnist