Update convolutional_mlp.py