added initial Python solution
added python output