add python script for problem 1