Added button to trigger logic