Update convolutional_mlp_squint.py