How To Square A Numpy Array