diff --git a/svg.cpp b/svg.cpp
index 9533773..88bf76a 100644
--- a/svg.cpp
+++ b/svg.cpp
@@ -29,7 +29,26 @@ void
 svg_rect(double x, double y, double width, double height, string stroke = "chartreuse", string fill = "plum"){
     cout << "<rect x='" << x << "' y='" << y << "' width='" << width << "' height='" << height << "' stroke='" << stroke << "' fill='" << fill << "'/>";
 }
-
+void
+input_colors(string &colors_stroke, string &colors_fill){
+    string color_personal;
+    cout << "Do you want to change colors? Yes/No" << endl;
+    cin >> color_personal;
+    while ((color_personal != "No") && (color_personal != "Yes")){
+        cout << "WRONG ANSWER! Try again!" << endl;
+        cin >> color_personal;
+    }
+    if (color_personal == "No"){
+        return;
+    }
+    else {
+        cout << "What color for a stroke do you want?" << endl;
+        cin >> colors_stroke;
+        cout << "What color to fill do you want?" << endl;
+        cin >> colors_fill;
+    }
+    return;
+}
 void
 show_histogram_svg(const vector<size_t>& bins, string stroke, string fill) {
     const auto IMAGE_WIDTH = 400;
diff --git a/unittest.cpp b/unittest.cpp
index 0acaa65..f4d7164 100644
--- a/unittest.cpp
+++ b/unittest.cpp
@@ -2,6 +2,23 @@
 #define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
 #include "doctest.h"
 #include "histogram_internal.h"
+#include "input_colors_internal.h"
+
+TEST_CASE("Mistakes in spelling") {
+    std::string colors_stroke = "black";
+    std::string colors_fill = "green";
+    input_colors(colors_stroke, colors_fill);
+    CHECK(colors_stroke == "yellow");
+    CHECK(colors_fill == "pink");
+}
+
+TEST_CASE("The answer is No") {
+    std::string colors_stroke = "black";
+    std::string colors_fill = "green";
+    input_colors(colors_stroke, colors_fill);
+    CHECK(colors_stroke == "black");
+    CHECK(colors_fill == "green");
+}
 
 TEST_CASE("distinct positive numbers") {
     double min = 0;