diff --git a/sca/ascad/attack.py b/sca/ascad/attack.py
index b9eb3ba..fd217b7 100644
--- a/sca/ascad/attack.py
+++ b/sca/ascad/attack.py
@@ -76,11 +76,7 @@ def align(db, ntraces):
 
 def main(db):
     ntraces = 1000
-    # traces = align(db, ntraces)
-
-    traces = numpy.zeros((ntraces, 700), 'double')
-    for i in range(ntraces):
-        traces[i, :] = db["traces"][i]
+    traces = align(db, ntraces)
 
     # for i in range(10):
     #     plt.plot(traces[i], color=numpy.hstack((numpy.random.random(3), [0.5])))
@@ -136,6 +132,5 @@ if __name__ == "__main__":
     from tqdm.contrib.logging import logging_redirect_tqdm
     logging.basicConfig(level=logging.INFO)
     with logging_redirect_tqdm():
-        # db = h5py.File("./ASCAD_databases/ASCAD_desync100.h5", "r")
-        db = h5py.File("./ASCAD_databases/ASCAD.h5", "r")
+        db = h5py.File("./ASCAD_databases/ASCAD_desync100.h5", "r")
         main(db["Attack_traces"])
diff --git a/sca/ascad/first_order.png b/sca/ascad/first_order.png
new file mode 100644
index 0000000..2acd3ff
Binary files /dev/null and b/sca/ascad/first_order.png differ