Hi there. The function > rlt(X, L_0=64, gamma=None, i_max=100) requires X to be 2D. For grayscale images like MNIST it works but how would you handle colour images (3D array) ? Thanks!