Python's homework