adding in theta step