Updated PID controller