add more parameter when saving
new training and init method that can pick file
new script