Bugfix to RESTART button, added comments to code