Calculate the square of a number using Python in WebAssembly