modify design

pull req