Forgotten files.
Moved train model creating once.