changed train_extension to train_extensions, so it actually looked in a file that exists