Added a restart-function