Was interested to analyze the wavemix block for single channel images. Please upload MNIST demo as experimented in the paper