added restart button