Added button and functionality