added python code used to create figures