Need some help in finding error in the simulation.

[the-morpheus21]

I was working on Simulation of Hydrogen atoms orbital simulation. I wrote a code for doing it with manim for a part of it.
Now I am getting an error ValueError: non-broadcastable output operand with shape (16,1) doesn't match the broadcast shape (16,3)
This is the code that I have written

`
from manim import *
import numpy as np
import scipy.special as spf

class HydroSimul(ThreeDScene): #phi = v theta = u
def psi_Y(self,v,u):

	  l=1
	  m=0
	  if m < 0 :
		  coeff1 = (-1)**m*np.sqrt((2*l+1)/4*np.pi)
		  coeff2 = np.sqrt(np.math.factorial(l-abs(m))/np.math.factorial(l+abs(m)))
		  coeff3 = np.sqrt(2)*np.sin(abs(m)*v)
		  leg = spf.lpmv(m,l,np.cos(u))
		  return np.array([coeff1* coeff2* coeff3* leg])

	  elif m == 0 :
		  coeff1 = (-1)**m*np.sqrt((2*l+1)/4*np.pi)
		  leg = spf.lpmv(m,l,np.cos(u))
		  return np.array([coeff1* leg])
	  
	  elif m > 0 :
		  coeff1 = (-1)**m*np.sqrt((2*l+1)/4*np.pi)
		  coeff2 = np.sqrt(np.math.factorial(l-m)/np.math.factorial(l+m))
		  coeff3 = np.sqrt(2)*np.cos(m*v)
		  leg = spf.lpmv(m,l,np.cos(u))
		  return np.array([coeff1* coeff2* coeff3* leg])
		  
  def construct(self):
	  axes = ThreeDAxes()
	  sphere = Surface(lambda u, v: self.psi_Y(u, v),v_range=[0, TAU], u_range=[-PI , PI ])
	  self.set_camera_orientation(phi=75 * DEGREES, theta=30 * DEGREES)
	  self.add(axes, sphere)`

Hope to solve it soon. Thanks in advance