updated the design
design fix