added python code used to create figure and print out data