renamed train_extentions to train_extensions