Added button to main