new python plot code added