Add button to set timebox while running