train mode works