2nd python practice problem