You could store the button press in EEPROM. Take a look at the example sketches in the IDE under ">File >Examples >EEPROM"