Add annotation to the get_sep_dist() function