More speed improvements with trainer
Train using last observation also