added solution for problem 3 in python