python plot fix