Small fix for sqrt

Use **0.5 instead of np.sqrt