Add end simulation hook