Now working with linear activations. Had to reduce learning rate to get rid of NaN