Добавил илюстрации работы и немного поменял код