First attempt at listening to buttons