A class Assignment for programming. (Python)