Added button functionality.