changed max epochs