Working backprop
Pinned down output format