Add reload handling logic