небольшие изменения в алгоритме поиска диагонали

небольшие изменения в алгоритме поиска диагонали