updated code to handle theta_2=0