added restart button (works)