Added restart button