Add button to restart. Implements, closes #2
It just works. Need to order the code and stuff.