modify design

modify design