Corrected figures and python script to generate them.