Added a restart and quit feature.