add reset button function + little edit