Add square root regression test