diff --git a/OpenOrca.py b/OpenOrca.py
index 92ba262..609fbe2 100644
--- a/OpenOrca.py
+++ b/OpenOrca.py
@@ -3,22 +3,6 @@ from datasets.features import Features, Value
 import json
 import os
 
-class CustomDataset(DatasetBuilder):
-    def _info(self) -> DatasetInfo:
-        return DatasetInfo(
-            features=Features({
-                'id': Value('string'),
-                'system_prompt': Value('string'),
-                'question': Value('string'),
-                'response': Value('string')
-            }),
-        )
-
-from datasets import DatasetBuilder, DatasetInfo, SplitGenerator, SplitInfo
-from datasets.features import Features, Value
-import json
-import os
-
 class CustomDataset(DatasetBuilder):
     def _info(self) -> DatasetInfo:
         return DatasetInfo(