#7 Added restart feature.