added python notebook for training