- Small modif to python plot