Adding back the button (I hope).