Diffusion Noise Propagation - Coding problem | OneNoughtOne