Still trying to fix the w(theta) section