diff --git a/main.cpp b/main.cpp
index edc3077..1eee826 100644
--- a/main.cpp
+++ b/main.cpp
@@ -2,6 +2,7 @@
 #include <conio.h>
 #include <vector>
 #include <stdlib.h>
+#include <string>
 #include "histogram.h"
 #include "text.h"
 #include "svg01.h"
@@ -10,6 +11,8 @@ using namespace std;
 struct Input {
     vector<double> numbers;
     size_t bin_count{};
+    string stroke;
+    string fill;
 };
 
 Input
@@ -30,6 +33,13 @@ input_data(){
 
     cerr << "Enter bin count: ";
     cin >> in.bin_count;
+
+    cerr << "Enter the color for the stroke: ";
+    cin >> in.stroke;
+
+    cerr << "Enter the color to fill: ";
+    cin >> in.fill;
+
     return in;
 }
 
@@ -38,6 +48,6 @@ int main()
 
     auto in = input_data();
     std::vector<size_t> bins = make_histogram(in.numbers, in.bin_count);
-    show_histogram_svg(bins);
+    show_histogram_svg(bins, in.stroke, in.fill);
     return 0;
 }