def square(N): i = np.square(N) return i