Added python solution for problem 3