Continued drying model development

Started to get results, but still does not perform as expected.