Continued development off drying model

Integrated diffusion into model