Added restart
Added a test