Some tuning of the PID. Also switched classes for it.