Update post_processing.py

+ added arguments to better plot the gaussian