Adding the button

Addition of the button, plus other bits