Trying to solve http://cognitivemedium.com/rmnist - View it on GitHub