solution

solution to problem 1 in python