diff --git a/src/main_2D.cpp b/src/main_2D.cpp
index 49878ab..fbcfaf5 100644
--- a/src/main_2D.cpp
+++ b/src/main_2D.cpp
@@ -32,9 +32,7 @@ int main(int argc, char *argv[]) {
 
   cout << setprecision(12);
 
-  // loop 100 times
-  // output is currently generated by the method itself
-  for (int t = 0; t < 1; t++) {
+  for (int t = 0; t < 10; t++) {
     diffu.simulate(field, alpha);
 
     cout << "Iteration: " << t << "\n\n";