Python List Difference Between Two Squares