Gradients properly calculating for MLP (I Hope)
Checkpoint, broken
Added test_mlp