ChatGPT - Modify code for button