Added the buttons to search-mode, unfortunately they need to be hardcoded and at the moment they're malplaced.