Adding functionality to our button