Added code for buttons