Каждый раз, когда чувствуется необходимость написать комментарий, лучше подумать, как бы переписать код так, чтобы он не был нужен.

В данном случае, `#write to file` намекает на необходимость функции `def write_to_file(distances)`.

Коммент `#bfs` выше сигнализирует о том, что надо вынести bfs в отдельную глобальную функцию, и избавится от 'глобальных' переменных `visited` и `distance`