Add Plotting in Python task