Minor update in main loop