Creating an array squared function using NumPy C-API