added python to generate graphs