fiddled with restart code