Theresa May makes joke about Boris Johnson staying 'on message