diff --git a/lab4/src/algo/images.rs b/lab4/src/algo/images.rs
index d484ca3..3badb06 100644
--- a/lab4/src/algo/images.rs
+++ b/lab4/src/algo/images.rs
@@ -55,13 +55,13 @@ pub fn gen_images() -> ([&'static str; 8], [[[bool; 7]; 7]; 8])  {
             // >=
             image(
                 [
-                    " *     ",
                     "  *    ",
                     "   *   ",
+                    "    *  ",
+                    "   *   ",
+                    "  * *  ",
+                    "   *   ",
                     "  *    ",
-                    " * *   ",
-                    "  *    ",
-                    " *     ",
                 ]
             ),
             // =
diff --git a/lab4/src/main.rs b/lab4/src/main.rs
index 4ece3c2..adf2224 100644
--- a/lab4/src/main.rs
+++ b/lab4/src/main.rs
@@ -89,7 +89,8 @@ impl eframe::App for MyApp {
                             epochs_left: self.epochs_count,
                             epochs_per_update: min(10, self.epochs_count / 60),
                             n: 0.0,
-                        }
+                        };
+                        self.model.set_random_weights(&mut rand::rng());
                     }
 
                     match &mut self.training {
@@ -180,5 +181,9 @@ impl eframe::App for MyApp {
                     });
                 });
         });
+
+        if matches!(self.training, TrainingState::Training {..}) {
+            ui.request_repaint()
+        }
     }
 }