From 16b2409ae81e01a5a63eb0b5297a736155b14523 Mon Sep 17 00:00:00 2001
From: Patryk Hegenberg <patrykhegenberg@gmail.com>
Date: Fri, 4 Jul 2025 10:57:09 +0200
Subject: [PATCH] refactor: update coding style

---
 cmd/main.go             |  2 +-
 internal/ui/training.go | 14 ++++++--------
 2 files changed, 7 insertions(+), 9 deletions(-)

diff --git a/cmd/main.go b/cmd/main.go
index b88d9a7..2fffb8f 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -29,7 +29,7 @@ func main() {
 	apiService := services.NewApiService(myApp.UniqueID())
 	trainingService := services.NewTrainingService(dbService, settingsService, apiService)
 
-	contentContainer := container.NewMax()
+	contentContainer := container.NewStack()
 	var navigateTo func(string)
 
 	trainingScreen, startTrainingAction := ui.MakeTrainingScreen(trainingService, settingsService, mainWindow)
diff --git a/internal/ui/training.go b/internal/ui/training.go
index 7050f69..5480a4d 100644
--- a/internal/ui/training.go
+++ b/internal/ui/training.go
@@ -75,15 +75,13 @@ func MakeTrainingScreen(ts *services.TrainingService, ss *services.SettingsServi
 		mainTimer = time.NewTicker(time.Second)
 		go func() {
 			for mainTimer != nil {
-				select {
-				case <-mainTimer.C:
-					if ts.State.RemainingSeconds <= 0 {
-						finishAction()
-						return
-					}
-					ts.Tick()
-					updateUI()
+				<-mainTimer.C
+				if ts.State.RemainingSeconds <= 0 {
+					finishAction()
+					return
 				}
+				ts.Tick()
+				updateUI()
 			}
 		}()
 	}