I have redone this and added a GUI for when you hit Reset, Apart from that, it's still the same.
This just came to me when I was bored.
