Restart button

Added restart button