Fix the X and Y axis ranges in the plots
Activate saving plots
Simplify plotting code