Starting using mpi4py
Switched Learn for-loops to make parallelization faster