PWM code fixes