Modelling of air gap membrane distillation and its application in heavy metals removal