updates for improved training