Fixed units in  `sqrt`

After filing issue #3 I took a closer look at the code and realized this was really easy to fix. So I fixed it =)