Small updates to reset button and py