Improved restart feature with a real button