Added python problem 2