Added features to PID controller