Python List Difference Between Two Squares Definition