diff --git a/fiab/extract.py b/fiab/extract.py
index 3af796c..232747d 100644
--- a/fiab/extract.py
+++ b/fiab/extract.py
@@ -1,5 +1,6 @@
 
 import json
+from collections import defaultdict
 
 metadata = json.load(open("raw_anemoi_metadata.json"))
 
@@ -15,8 +16,16 @@ pressure_level_variables = [v for v in variables if '_' in v]
 pressure_levels = sorted(set([v.split('_')[-1] for v in pressure_level_variables]))
 pressure_level_variables = sorted(set([v.split('_')[0] for v in pressure_level_variables]))
 
+levels_for_variables = defaultdict(list)
+for v in variables:
+    if "_" in v:
+        variable, level = v.split('_')
+        levels_for_variables[variable].append(level)
+
+print('Levels for variables:', levels_for_variables)
+
 print('Pressure level variables:', pressure_level_variables)
-print('Pressure levels:', pressure_levels)
+print('Pressure levels:', sorted([int(p) for p in pressure_levels]))
 
 print('Surface variables:', surface_variables)