Added Python solution for problem 2