added problem 1 python solution