Restart feature added