Double Squares (CodeEval)