A function skeleton for the square function
Finish the square function