Focus restore modal buttons