added python solution
added python solution output