added python solution for prob1