convolutional_mlp.py modified