updated README and added python code