added python code example