Added warm restart code