Modelling the spread of disease