def Rz_matrix(theta):
return np.array([<EOL>[np.cos(theta), -np.sin(theta), <NUM_LIT:0>],<EOL>[np.sin(theta), np.cos(theta), <NUM_LIT:0>],<EOL>[<NUM_LIT:0>, <NUM_LIT:0>, <NUM_LIT:1>]<EOL>])<EOL>
Rotation matrix around the Z axis