Added function to check thetas and tried to color the x,y,z axes for the plot.