Fixed what refactoring broke
Added heat diffusion and buoyancy model