Added restart menu