Fix AO zero correction mistake
Improved AO training (still theoretical)