Add test for extra sqrt functionality