fixing up the design