implemented restart button