Added a reset button and some other stuff