Implemented optimized diffusion code in starburst