Added code for the buttons