Update and rename add_button to add_button_working