diff --git a/MPI/main.cpp b/MPI/main.cpp
index 4a9a924..37d3432 100644
--- a/MPI/main.cpp
+++ b/MPI/main.cpp
@@ -1,75 +1,72 @@
 #include <iostream>
-#include <vector>
 #include <chrono>
 #include <mpi.h>
-#include <iomanip>
 
 #include "Simulation.h"
-#include "const.h"
 
 int main(int argc, char *argv[]) {
     if (argc < 9) {
         std::cerr << "Usage: " << argv[0] << "-i <data folder> -nx <size of problem> -t <end time of simulation> \n"
                                              "-o <output folder> [-r : init. with random values] \n\n";
         return 1;
     }
 
     std::string input_folder;
     std::string output_folder;
     int nx;
     double tend;
     bool random_values = false;
     std::string flag;
 
     //Assigning all the correct arguments
     for(int i = 1;i < argc;++i) {
         if(std::string(argv[i]) == "-o") {
             flag = "o";
             continue;
         }
         else if(std::string(argv[i]) == "-i") {
             flag = "i";
             continue;
         }
         else if(std::string(argv[i]) == "-nx") {
             flag = "nx";
             continue;
         }
         else if(std::string(argv[i]) == "-t") {
             flag = "t";
             continue;
         }
         else if(std::string(argv[i]) == "-r") {
             random_values = true;
             continue;
         }
 
         if(flag == "o")
             output_folder = std::string(argv[i]);
         else if(flag == "i")
             input_folder = std::string(argv[i]);
         else if(flag == "nx")
             nx = std::stoi(argv[i]);
         else if(flag == "t")
             tend = std::stod(argv[i]);
     }
 
     MPI_Init(&argc, &argv);
 
     int prank, psize;
     MPI_Comm_rank(MPI_COMM_WORLD, &prank);
     MPI_Comm_size(MPI_COMM_WORLD, &psize);
 
     Simulation sim(input_folder, nx, 500, tend, true, random_values);
 
     sim.compute();
 
     if(prank == 0)
         std::cout << "Perf. = " << sim.get_flops_performance()/1e9 << "Gflops / TTS : " << sim.get_TTS()/1e3 << "s" << std::endl;
 
     std::string file_name = output_folder + "Data_nx" + std::to_string(nx) + "_500km_T0.2_h_Sol.bin";
     sim.save(file_name);
 
     MPI_Finalize();
     return 0;
-}
\ No newline at end of file
+}