Add PWM functionality