Added possibility to restart