Python Data Structures Assignment 940