added a RESTART button and some minor changes