Add support for restart process