Added Python solution for problem 6