Major rewrite:
0) Added Convolutional layer
1) Made the code Object Oriented
2) Better printing
3) Checking for inf cost
4) Using probabilities from last two labels (as opposed to just the last)