further improvements and code refactor; appears to be working on mnist and simple regression
continued updates
mnist example