diff --git a/Day01/main.go b/Day01/main.go
index b1e9a2f..516b458 100644
--- a/Day01/main.go
+++ b/Day01/main.go
@@ -11,8 +11,7 @@ func main() {
 	fmt.Printf("|| AdventOfCode 2024 - Day %02d ||\n", DAY)
 	fmt.Println("||++++++++++++++++++++++++++++||\n")
 
-	fmt.Println("Original Code")
 	original()
-	fmt.Println("\nOptimized Code by ChatGPT")
+	fmt.Println()
 	optimized()
 }
diff --git a/Day01/optimized.go b/Day01/optimized.go
index 1c5b789..8e5a6ef 100644
--- a/Day01/optimized.go
+++ b/Day01/optimized.go
@@ -9,6 +9,8 @@ import (
 )
 
 func optimized() {
+	fmt.Println("Optimized Code by ChatGPT")
+
 	// Read the file in one step
 	content, err := os.ReadFile("./input.json")
 	if err != nil {
diff --git a/Day01/original.go b/Day01/original.go
index 63c691f..d9225d9 100644
--- a/Day01/original.go
+++ b/Day01/original.go
@@ -9,6 +9,8 @@ import (
 )
 
 func original() {
+	fmt.Println("Original Code")
+
 	content, err := ioutil.ReadFile("./input.json")
 	if err != nil {
 		log.Fatal("Error when opening input.json: ", err)