diff --git a/TWOPO.cpp b/TWOPO.cpp
index bb8cd72..7008899 100644
--- a/TWOPO.cpp
+++ b/TWOPO.cpp
@@ -14,21 +14,22 @@ struct Input {
 };
 
 //ввод
-Input input_data() {
+Input
+input_data(istream& inp) {
     size_t number_count;
     cerr << "Введите количество оценок: ";
-    cin >> number_count;
+    inp >> number_count;
 
     Input in;
     in.numbers.resize(number_count);
 
     cerr << "Введите оценки:\n";
     for (size_t i = 0; i < number_count; i++) {
-        cin >> in.numbers[i];
+        inp >> in.numbers[i];
     }
 
     cerr << "Введите количество корзин: ";
-    cin >> in.bin_count;
+    inp >> in.bin_count;
 
     return in;
 }
@@ -36,7 +37,7 @@ Input input_data() {
 int main() {
     setlocale(LC_ALL, "Russian");
 
-    auto in = input_data();
+    auto in = input_data(cin);
 
     auto bins = make_histogram(in.numbers, in.bin_count);