Added a test for square root