add some button functionality