diff --git a/src/h3_stage_2.py b/src/h3_stage_2.py
index bca6470..1fe7f1d 100644
--- a/src/h3_stage_2.py
+++ b/src/h3_stage_2.py
@@ -1,7 +1,8 @@
-import pandas as pd
+from concurrent.futures import ThreadPoolExecutor, wait
+
 import numpy as np
+import pandas as pd
 import tensorflow as tf
-from concurrent.futures import ThreadPoolExecutor, wait
 
 # Concurrency stuff
 pool = ThreadPoolExecutor(20)