use python instead of python-mpi because python-mpi was causing mem issues