updated for long training runs