introduced epochs into training