Add propagation model to code