def apply_threshold(heatmap, threshold):
    """
    Apply threshold to the heatmap to remove false positives.
        Parameters:
            heatmap: Input heatmap.
            threshold: Selected threshold.
    """
    heatmap[heatmap < threshold] = 0 
    return heatmap