working on restart button