Change a couple of things around with the button