Added a restart button
Merge pull request #17 from zael99/master

Added a restart button