diff --git a/main.cpp b/main.cpp
index dbb0370..125747c 100644
--- a/main.cpp
+++ b/main.cpp
@@ -12,18 +12,27 @@ struct Input {
     size_t block_width{};
 };
 
-Input input_data(istream& in_steam = cin) {
-    cerr << "Enter number count: ";
-    cin >> number_count;
+Input input_data(istream& in_steam = cin, bool prompt = true) {
+    size_t number_count;
+    if (prompt) {
+            cerr << "Enter number count: ";
+    }
+    in_steam >> number_count;
 
     Input in;
     in.numbers.resize(number_count);
 
+    if (prompt) {
+        cerr << "Enter " << number_count << " numbers: ";
+    }
     for (size_t i = 0; i < number_count; i++) {
-        cin >> in.numbers[i];
+        in_stream >> in.numbers[i];
+    }
+
+    if (prompt) {
+        cerr << "Enter bin count: ";
     }
-    cerr << "Enter bin count: ";
-    cin >> in.bin_count;
+    in_stream >> in.bin_count;
 
     return in;
 }