Added problem 1+2 in python