Added PID controller