diff --git a/dataset.py b/dataset.py
index 04ea578..52220c0 100644
--- a/dataset.py
+++ b/dataset.py
@@ -3,6 +3,18 @@ from enum import Enum
 from typing import Any, Dict, List, Optional, Union
 import json, argparse, os, uuid
 
+
+class OutputType(Enum):
+    FULL = "full"
+    CODE_REFINEMENT = "code_refinement"
+    COMMENT_GEN = "comment_gen"
+
+
+class ArchiveState(Enum):
+    BASE = "base"
+    MERGED = "merged"
+
+
 # fmt: off
 @dataclass
 class FileData:
@@ -25,10 +37,6 @@ class Selection:
     diff_after_address_change: Optional[bool]
     is_code_related: bool
 
-class ArchiveState(Enum):
-    BASE = "base"
-    MERGED = "merged"
-
 @dataclass
 class Metadata:
     id: str
@@ -88,11 +96,6 @@ class CodeRefinementEntry:
             comments=entry.comments,
         )
 
-class OutputType(Enum):
-    FULL = "full"
-    CODE_REFINEMENT = "code_refinement"
-    COMMENT_GEN = "comment_gen"
-
 # fmt: on
 @dataclass
 class Dataset: