python problem 3 added