Remove the axis of the imshow plots
Move most of code in "if __name__" to a function

This allows better reuse of the simulation code in the
simulate_metis_scenario.py in the corresponding IPython notebook.