MITx: 6.00x Python code