Improve the OK button