added python solution for problem set 6