Add class to handle the heatmap and the build slice.