TEKO Vietnam interview question

Write a function to calculate square root