add ability to restart