Added a button (crazy, I know)