fixed python plot example