Added problem 1 & 2 python