Added basic button functionality