Implemented restart functionality.