Added button and click behavior