update 2_classifying_mnist_using_lr