Added linkage to the buttons