Основное закончил, теперь сокращаем код