Set a button for testing UI