You have to apply the same slicing to the angle_rads that are inside the np.sin() call.
It is solved now! Thank you so much!!!