added a restart button that only works after you have already started