Adding mpi_config.py and main file for testing