3D diffusion now uses saved cross-sections for initial state