add restartController function