Proper module definition of ufunc and gufunc in vectorize