Add button to test function.