fixing up the design
more design