Updated Models and Heatmap