Add python solutions for problem 1 and 2