Add end-of-round logic