fixed training - I think
think training works now...