Added solution for problem 1 in python