Adding button functionality.