fixing design
design changes