Add small restart script