fixed python script - diff eq.