Added a couple of buttons