Add a button for testing