Update Python Problem 301