Upped number of epochs again