Change update main loop