Added a restart button but not working, yet.