Updated model to include distance.