added python plotting