Trying to add funtionality to buttons
.