Improved Diffusion Github Student Plan