Also changed the ok button