added python version of plot making