1 files changed, 9 insertions, 7 deletions
diff --git a/src/Day01.kt b/src/Day01.kt
index 6d45527..4ac1ef7 100644
--- a/src/Day01.kt
+++ b/src/Day01.kt
@@ -1,22 +1,24 @@
-fun main() {
-    fun part1(input: List<Int>): Int =
+object Day01 {
+    fun part1(input: List<Int>) =
         input
             .zipWithNext()
             .count { it.second > it.first }
 
-    fun part2(input: List<Int>): Int =
+    fun part2(input: List<Int>) =
         input
             .asSequence()
             .windowed(3)
             .map { it.sum() }
             .zipWithNext()
             .count { it.second > it.first }
+}
 
+fun main() {
     val testInput = readInputAsNumbers("Day01_test")
-    check(part1(testInput) == 7)
-    check(part2(testInput) == 5)
+    check(Day01.part1(testInput) == 7)
+    check(Day01.part2(testInput) == 5)
 
     val input = readInputAsNumbers("Day01")
-    println(part1(input))
-    println(part2(input))
+    println(Day01.part1(input))
+    println(Day01.part2(input))
 }