better variable names in train()