Now coloring by the amount of infections using custom color map. #211