diff --git a/project/main.cpp b/project/main.cpp
index 717dfb8..efafb70 100644
--- a/project/main.cpp
+++ b/project/main.cpp
@@ -12,9 +12,10 @@ struct Input
 };
 
 Input
-input_data(istream& tin)
+input_data(istream& tin, bool prompt)
 {
     size_t number_count;
+    if (prompt)
     cerr << "Enter number count: ";
     tin >> number_count;
     Input in;
@@ -23,13 +24,16 @@ input_data(istream& tin)
     {
         tin >> in.numbers[i];
     }
+    if (prompt)
+    {
     cerr << "Enter bin count: ";
+    }
     tin >> in.bin_count;
     return in;
 }
 int main()
 {
-    Input in = input_data(cin);
+    Input in = input_data(cin,true);
     auto bins = make_histogram(in.numbers, in.bin_count);
     show_histogram_svg(bins);
     return 0;