edits for theta_set function