Added button and listener