Add function to set save trainer name