Can you somehow move this at the top as a constant (e.g.: `LABEL_TO_FILENAMES` above) or helper function.