Python difference of squares exer done